package main

/*
Go支持sqlite的驱动也比较多，但是好多都是不支持database/sql接口的

https://github.com/mattn/go-sqlite3 支持database/sql接口，基于cgo(关于cgo的知识请参看官方文档或者本书后面的章节)写的
https://github.com/feyeleanor/gosqlite3 不支持database/sql接口，基于cgo写的
https://github.com/phf/go-sqlite3 不支持database/sql接口，基于cgo写的
目前支持database/sql的SQLite数据库驱动只有第一个，我目前也是采用它来开发项目的。采用标准接口有利于以后出现更好的驱动的时候做迁移。
*/
import (
	"database/sql"
	"fmt"
	_ "github.com/mattn/go-sqlite3"
	"time"
)

func main() {
	db, err := sql.Open("sqlite3", "./godb.db")
	checkErr(err)

	// 插入数据
	stmt, err := db.Prepare("INSERT INTO userinfo(username,department,created) VALUES (?,?,?)")
	checkErr(err)

	res, err := stmt.Exec("dralee", "研发部门", time.Now())
	checkErr(err)

	id, err := res.LastInsertId()
	checkErr(err)
	fmt.Println("新增的id:", id)

	// 更新数据
	stmt, err = db.Prepare("UPDATE userinfo set username=? WHERE uid=?")
	checkErr(err)

	res, err = stmt.Exec("dralee2", id)
	checkErr(err)

	affect, err := res.RowsAffected()
	checkErr(err)
	fmt.Println("更新影响行数：", affect)

	// 查询数据
	rows, err := db.Query("SELECT * FROM userinfo")
	checkErr(err)

	for rows.Next() {
		var uid int
		var username string
		var department string
		var created time.Time
		err = rows.Scan(&uid, &username, &department, &created)
		checkErr(err)
		fmt.Println("uid:", uid)
		fmt.Println("username:", username)
		fmt.Println("department:", department)
		fmt.Println("created:", created)
	}
	// 删除数据
	stmt, err = db.Prepare("DELETE FROM userinfo WHERE uid=?")
	checkErr(err)

	res, err = stmt.Exec(id)
	checkErr(err)

	affect, err = res.RowsAffected()
	checkErr(err)

	fmt.Println("删除影响行数：", affect)
	db.Close()
}

func checkErr(err error) {
	if err != nil {
		panic(err)
	}
}
